/*
 * Copyright (c) 2011-2022, baomidou (jobob@qq.com).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.sky.wsp.mybatis.plus.plugins.handler;

import net.sf.jsqlparser.schema.Column;

import java.util.List;

/**
 * 参考MybatisPlus（版本3.4.0）的多租户处理器，改造成本数据权限处理器
 *
 * @author wangshaopeng@talkweb.com.cn
 * @Date 2023-01-11
 * @see com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler
 */
public interface MyDataPermissionHandler {

    /**
     * 获取数据创建人
     * <p>
     *
     * @return 获取数据创建人
     */
    Long getDataCreator();

    /**
     * 获取数据创建人字段名
     * <p>
     * 默认字段名叫: create_by
     *
     * @return 数据创建人字段名
     */
    default String getDataCreatorColumn() {
        return "creator";
    }

    /**
     * 获取数据权限 ID 集合
     * <p>
     *
     * @return 数据权限 ID 集合
     */
    List<Long> getDataPermissionIdSet();

    /**
     * 获取数据权限字段名
     * <p>
     * 默认字段名叫: data_permission_id
     *
     * @return 数据权限字段名
     */
    default String getDataPermissionIdColumn() {
        return "data_permission_id";
    }

    /**
     * 根据表名判断是否忽略拼接多数据权限条件
     * <p>
     * 默认都要进行解析并拼接多数据权限条件
     *
     * @param tableName 表名
     * @return 是否忽略, true:表示忽略，false:需要解析并拼接多数据权限条件
     */
    default boolean ignoreTable(String tableName) {
        return false;
    }

    /**
     * 忽略插入数据权限字段逻辑
     *
     * @param columns                插入字段
     * @param dataPermissionIdColumn 数据权限 ID 字段
     * @return
     */
    default boolean ignoreInsert(List<Column> columns, String dataPermissionIdColumn) {
        return columns.stream().map(Column::getColumnName).anyMatch(i -> i.equalsIgnoreCase(dataPermissionIdColumn));
    }
}
